from django.shortcuts import render

# Create your views here.
from django.views import View
from django import http
from apps.areas.models import Area


class SubAreasView(View):

    def get(self,request,pk):

        try:
            # 查询市区数据
            sub_model_list = Area.objects.filter(parent=pk)
            parent_model = Area.objects.get(id=pk)

            sub_list = []
            for sub_model in sub_model_list:
                sub_list.append({'id':sub_model.id,
                                 'name':sub_model.name})
            sub_data = {
                'id': parent_model.id,
                'name': parent_model.name,
                'subs': sub_list
            }
        except Exception as e:
            return http.JsonResponse({'code':400,'errmsg':'数据错误'})

        return http.JsonResponse({'code':0,'errmsg':'ok','sub_data':sub_data})



class ProvinceAreasView(View):

    def get(self,request):

        try:
            # 查询省级数据
            province_model_list = Area.objects.filter(parent_id__isnull=True)

            province_list = []
            for province_model in province_model_list:
                province_list.append({'id':province_model.id,
                                      'name': province_model.name})

        except Exception as e:
            print(e)
            return http.JsonResponse({'code':400,'errmsg':'省份数据错误'})

        return http.JsonResponse({'code':0,'errmsg':'ok','province_list':province_list})
